# -*- coding: utf-8 -*-
# @Time        :2024/9/27 下午5:54
# @Author      :文刀水寿
# @File        : demo36.py
"""
 @Description :演示可视化需求1
"""
import json
from pyecharts.charts import Line
from pyecharts.options import TitleOpts, LabelOpts

# 处理数据
f_us = open("D:/美国.txt", "r", encoding='UTF-8')
us_data = f_us.read()

f_jp = open("D:/日本.txt", 'r', encoding='UTF-8')
jp_data = f_jp.read()

f_in = open("D:/印度.txt", 'r', encoding='UTF-8')
in_data = f_in.read()

# 去掉不合json规范的开头
us_data = us_data.replace("jsonp_1629344292311_69436(", " ")
jp_data = jp_data.replace("jsonp_1629350871167_29498(", " ")
in_data = in_data.replace("jsonp_1629350745930_63180(", " ")

# 去掉不合json规范的结尾
us_data = us_data[:-2]
jp_data = jp_data[:-2]
in_data = in_data[:-2]

# json转Python字典
us_dict = json.loads(us_data)
jp_dict1 = json.loads(jp_data)
in_dict2 = json.loads(in_data)

# 获取trend key
us_trend_data = us_dict['data'][0]["trend"]
jp_trend_data = jp_dict1['data'][0]["trend"]
in_trend_data = in_dict2['data'][0]["trend"]

# 获取日期数据，用于x轴，取2020年（到314下标结束）
x_data = us_trend_data["updateDate"][:314]

# 获取确认数据，用于y轴，取2020年（到315下标结束）
us_y_data = us_trend_data["list"][0]["data"][:314]
jp_y_data = jp_trend_data["list"][0]["data"][:314]
in_y_data = in_trend_data["list"][0]["data"][:314]

# 生成图表
line = Line()
line.add_xaxis(x_data)

line.add_yaxis("美国死亡人数", us_y_data, label_opts=LabelOpts(is_show=False))
line.add_yaxis("日本死亡人数", jp_y_data, label_opts=LabelOpts(is_show=False))
line.add_yaxis("印度死亡人数", in_y_data, label_opts=LabelOpts(is_show=False))

# 全局设置
line.set_global_opts(
    title_opts=TitleOpts(title='2020美日印三国疫情死亡人数', pos_left="center", pos_bottom="1%")

)
line.render()
f_in.close()
f_jp.close()
f_us.close()
